
import pygame
from nation import Nation
from itertools import cycle

class GameLogic(object):

    def __init__(self):
        self.names = ['russia', 'eua', 'brasil', 'portugal', 'suecia', 'austria', 'holanda']
        self.numberOfNations = len(self.names)
        self.nations = [ Nation(index, name) for (index, name) in enumerate(self.names) ]
        self.mode = 'select'
        self.iterator_nation = cycle(enumerate(self.nations))
        self.current_nation_index, self.current_nation = self.iterator_nation.next()

    def next_turn(self):
        self.current_nation_index, self.current_nation = self.iterator_nation.next()
        return self.current_nation_index, self.current_nation

